home *** CD-ROM | disk | FTP | other *** search
/ Mac Easy 2010 May / Mac Life Ubuntu.iso / casper / filesystem.squashfs / usr / share / pyshared / DistUpgrade / DistUpgradeViewNonInteractive.py < prev    next >
Encoding:
Python Source  |  2009-04-27  |  12.5 KB  |  322 lines
  1. # DistUpgradeView.py 
  2. #  
  3. #  Copyright (c) 2004,2005 Canonical
  4. #  
  5. #  Author: Michael Vogt <michael.vogt@ubuntu.com>
  6. # 
  7. #  This program is free software; you can redistribute it and/or 
  8. #  modify it under the terms of the GNU General Public License as 
  9. #  published by the Free Software Foundation; either version 2 of the
  10. #  License, or (at your option) any later version.
  11. # 
  12. #  This program is distributed in the hope that it will be useful,
  13. #  but WITHOUT ANY WARRANTY; without even the implied warranty of
  14. #  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  15. #  GNU General Public License for more details.
  16. # 
  17. #  You should have received a copy of the GNU General Public License
  18. #  along with this program; if not, write to the Free Software
  19. #  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307
  20. #  USA
  21.  
  22. import apt
  23. import logging
  24. import time
  25. import sys
  26. from DistUpgradeView import DistUpgradeView, InstallProgress, FetchProgress
  27. from DistUpgradeConfigParser import DistUpgradeConfig
  28. import os
  29. import pty
  30. import apt_pkg
  31. import select
  32. import fcntl
  33. import string
  34. import re
  35. import subprocess
  36. from subprocess import call, PIPE, Popen
  37. import copy
  38. import apt.progress
  39. from ConfigParser import NoSectionError, NoOptionError
  40.  
  41. class NonInteractiveFetchProgress(FetchProgress):
  42.     def updateStatus(self, uri, descr, shortDescr, status):
  43.         FetchProgress.updateStatus(self, uri, descr, shortDescr, status)
  44.         #logging.debug("Fetch: updateStatus %s %s" % (uri, status))
  45.         if status == apt.progress.FetchProgress.dlDone:
  46.             print "fetched %s (%.2f/100) at %sb/s" % (uri, self.percent, 
  47.                                                       apt_pkg.SizeToStr(int(self.currentCPS)))
  48.             if sys.stdout.isatty():
  49.                 sys.stdout.flush()
  50.         
  51.  
  52. class NonInteractiveInstallProgress(InstallProgress):
  53.     def __init__(self):
  54.         InstallProgress.__init__(self)
  55.         logging.debug("setting up environ for non-interactive use")
  56.         os.environ["DEBIAN_FRONTEND"] = "noninteractive"
  57.         os.environ["APT_LISTCHANGES_FRONTEND"] = "none"
  58.         os.environ["RELEASE_UPRADER_NO_APPORT"] = "1"
  59.         self.config = DistUpgradeConfig(".")
  60.         try:
  61.             if self.config.getboolean("NonInteractive","ForceOverwrite"):
  62.                 apt_pkg.Config.Set("DPkg::Options::","--force-overwrite")
  63.         except (NoSectionError, NoOptionError), e:
  64.             pass
  65.         # more debug
  66.         #apt_pkg.Config.Set("Debug::pkgOrderList","true")
  67.         #apt_pkg.Config.Set("Debug::pkgDPkgPM","true")
  68.         # default to 2400 sec timeout
  69.         self.timeout = 2400
  70.         try:
  71.             self.timeout = self.config.getint("NonInteractive","TerminalTimeout")
  72.         except Exception, e:
  73.             pass
  74.     
  75.     def error(self, pkg, errormsg):
  76.         # re-run maintainer script with sh -x/perl debug to get a better 
  77.         # idea what went wrong
  78.         # 
  79.         # FIXME: this is just a approximation for now, we also need
  80.         #        to pass:
  81.         #        - a version after remove (if upgrade to new version)
  82.         #
  83.         #        not everything is a shell or perl script
  84.         #
  85.         # if the new preinst fails, its not yet in /var/lib/dpkg/info
  86.         # so this is inaccurate as well
  87.         logging.error("got a error from dpkg for pkg: '%s': '%s'" % (pkg, errormsg))
  88.         environ = copy.copy(os.environ)
  89.         environ["PYCENTRAL"] = "debug"
  90.         cmd = []
  91.  
  92.         # find what maintainer script failed
  93.         if "post-installation" in errormsg:
  94.             prefix = "/var/lib/dpkg/info/"
  95.             name = "postinst"
  96.             argument = "configure"
  97.             maintainer_script = "%s/%s.%s" % (prefix, pkg, name)
  98.         elif "pre-installation" in errormsg:
  99.             prefix = "/var/lib/dpkg/tmp.ci/"
  100.             #prefix = "/var/lib/dpkg/info/"
  101.             name = "preinst"
  102.             argument = "install"
  103.             maintainer_script = "%s/%s" % (prefix, name)
  104.         elif "pre-removal" in errormsg:
  105.             prefix = "/var/lib/dpkg/info/"
  106.             name = "prerm"
  107.             argument = "remove"
  108.             maintainer_script = "%s/%s.%s" % (prefix, pkg, name)
  109.         elif "post-removal" in errormsg:
  110.             prefix = "/var/lib/dpkg/info/"
  111.             name = "postrm"
  112.             argument = "remove"
  113.             maintainer_script = "%s/%s.%s" % (prefix, pkg, name)
  114.         else:
  115.             print "UNKNOWN (trigger?) dpkg/script failure for %s (%s) " % (pkg, errormsg)
  116.             return
  117.  
  118.         # find out about the interpreter
  119.         if not os.path.exists(maintainer_script):
  120.             logging.error("can not find failed maintainer script '%s' " % maintainer_script)
  121.             return
  122.         interp = open(maintainer_script).readline()[2:].strip().split()[0]
  123.         if ("bash" in interp) or ("/bin/sh" in interp):
  124.             debug_opts = ["-ex"]
  125.         elif ("perl" in interp):
  126.             debug_opts = ["-d"]
  127.             environ["PERLDB_OPTS"] = "AutoTrace NonStop"
  128.         else:
  129.             logging.warning("unknown interpreter: '%s'" % interp)
  130.  
  131.         # check if debconf is used and fiddle a bit more if it is
  132.         if ". /usr/share/debconf/confmodule" in open(maintainer_script).read():
  133.             environ["DEBCONF_DEBUG"] = "developer"
  134.             environ["DEBIAN_HAS_FRONTEND"] = "1"
  135.             interp = "/usr/share/debconf/frontend"
  136.             debug_opts = ["sh","-ex"]
  137.  
  138.         # build command
  139.         cmd.append(interp)
  140.         cmd.extend(debug_opts)
  141.         cmd.append(maintainer_script)
  142.         cmd.append(argument)
  143.  
  144.         # check if we need to pass a version
  145.         if name == "postinst":
  146.             version = Popen("dpkg-query -s %s|grep ^Config-Version" % pkg,shell=True, stdout=PIPE).communicate()[0]
  147.             if version:
  148.                 cmd.append(version.split(":",1)[1].strip())
  149.         elif name == "preinst":
  150.             pkg = os.path.basename(pkg)
  151.             pkg = pkg.split("_")[0]
  152.             version = Popen("dpkg-query -s %s|grep ^Version" % pkg,shell=True, stdout=PIPE).communicate()[0]
  153.             if version:
  154.                 cmd.append(version.split(":",1)[1].strip())
  155.  
  156.         print cmd
  157.         logging.debug("re-running '%s' (%s)" % (cmd, environ))
  158.         ret = subprocess.call(cmd, env=environ)
  159.         logging.debug("%s script returned: %s" % (name,ret))
  160.         
  161.     def conffile(self, current, new):
  162.         logging.warning("got a conffile-prompt from dpkg for file: '%s'" % current)
  163.         # looks like we have a race here *sometimes*
  164.         time.sleep(5)
  165.     try:
  166.           # don't overwrite
  167.       os.write(self.master_fd,"n\n")
  168.      except Exception, e:
  169.       logging.error("error '%s' when trying to write to the conffile"%e)
  170.  
  171.     def startUpdate(self):
  172.         InstallProgress.startUpdate(self)
  173.         self.last_activity = time.time()
  174.           
  175.     def updateInterface(self):
  176.         if self.statusfd == None:
  177.             return
  178.  
  179.         if (self.last_activity + self.timeout) < time.time():
  180.             logging.warning("no activity %s seconds (%s) - sending ctrl-c" % (self.timeout, self.status))
  181.             # ctrl-c
  182.             os.write(self.master_fd,chr(3))
  183.  
  184.  
  185.         # read status fd from dpkg
  186.         # from python-apt/apt/progress.py (but modified a bit)
  187.         # -------------------------------------------------------------
  188.         res = select.select([self.statusfd],[],[],0.1)
  189.         while len(res[0]) > 0:
  190.             self.last_activity = time.time()
  191.             while not self.read.endswith("\n"):
  192.                 self.read += os.read(self.statusfd.fileno(),1)
  193.             if self.read.endswith("\n"):
  194.                 s = self.read
  195.                 #print s
  196.                 (status, pkg, percent, status_str) = string.split(s, ":")
  197.                 if status == "pmerror":
  198.                     self.error(pkg,status_str)
  199.                 elif status == "pmconffile":
  200.                     # we get a string like this:
  201.                     # 'current-conffile' 'new-conffile' useredited distedited
  202.                     match = re.compile("\s*\'(.*)\'\s*\'(.*)\'.*").match(status_str)
  203.                     if match:
  204.                         self.conffile(match.group(1), match.group(2))
  205.                 elif status == "pmstatus":
  206.                     if (float(percent) != self.percent or 
  207.                         status_str != self.status):
  208.                         self.statusChange(pkg, float(percent), status_str.strip())
  209.                         self.percent = float(percent)
  210.                         self.status = string.strip(status_str)
  211.                         sys.stdout.write("[%s] %s: %s\n" % (float(percent), pkg, status_str.strip()))
  212.                         sys.stdout.flush()
  213.             self.read = ""
  214.             res = select.select([self.statusfd],[],[],0.1)
  215.         # -------------------------------------------------------------
  216.  
  217.         #fcntl.fcntl(self.master_fd, fcntl.F_SETFL, os.O_NDELAY)
  218.         # read master fd (terminal output)
  219.         res = select.select([self.master_fd],[],[],0.1)
  220.         while len(res[0]) > 0:
  221.            self.last_activity = time.time()
  222.            try:
  223.                s = os.read(self.master_fd, 1)
  224.                sys.stdout.write("%s" % s)
  225.            except OSError,e:
  226.                # happens after we are finished because the fd is closed
  227.                return
  228.            res = select.select([self.master_fd],[],[],0.1)
  229.         sys.stdout.flush()
  230.  
  231.     def fork(self):
  232.         logging.debug("doing a pty.fork()")
  233.         # some maintainer scripts fail without
  234.         os.environ["TERM"] = "dumb"
  235.         (self.pid, self.master_fd) = pty.fork()
  236.         if self.pid != 0:
  237.             logging.debug("pid is: %s" % self.pid)
  238.         return self.pid
  239.         
  240.  
  241. class DistUpgradeViewNonInteractive(DistUpgradeView):
  242.     " non-interactive version of the upgrade view "
  243.     def __init__(self, datadir=None, logdir=None):
  244.         self.config = DistUpgradeConfig(".")
  245.         self._fetchProgress = NonInteractiveFetchProgress()
  246.         self._installProgress = NonInteractiveInstallProgress()
  247.         self._opProgress = apt.progress.OpProgress()
  248.         sys.__excepthook__ = self.excepthook
  249.     def excepthook(self, type, value, traceback):
  250.         " on uncaught exceptions -> print error and reboot "
  251.         logging.error("got exception '%s': %s " % (type, value))
  252.         #sys.excepthook(type, value, traceback)
  253.         self.confirmRestart()
  254.     def getOpCacheProgress(self):
  255.         " return a OpProgress() subclass for the given graphic"
  256.         return self._opProgress
  257.     def getFetchProgress(self):
  258.         " return a fetch progress object "
  259.         return self._fetchProgress
  260.     def getInstallProgress(self, cache=None):
  261.         " return a install progress object "
  262.         return self._installProgress
  263.     def updateStatus(self, msg):
  264.         """ update the current status of the distUpgrade based
  265.             on the current view
  266.         """
  267.         pass
  268.     def setStep(self, step):
  269.         """ we have 5 steps current for a upgrade:
  270.         1. Analyzing the system
  271.         2. Updating repository information
  272.         3. Performing the upgrade
  273.         4. Post upgrade stuff
  274.         5. Complete
  275.         """
  276.         pass
  277.     def confirmChanges(self, summary, changes, downloadSize,
  278.                        actions=None, removal_bold=True):
  279.         DistUpgradeView.confirmChanges(self, summary, changes, downloadSize, actions)
  280.     logging.debug("toinstall: '%s'" % self.toInstall)
  281.         logging.debug("toupgrade: '%s'" % self.toUpgrade)
  282.         logging.debug("toremove: '%s'" % self.toRemove)
  283.         return True
  284.     def askYesNoQuestion(self, summary, msg):
  285.         " ask a Yes/No question and return True on 'Yes' "
  286.         return True
  287.     def confirmRestart(self):
  288.         " generic ask about the restart, can be overridden "
  289.     logging.debug("confirmRestart() called")
  290.         # ignore if we don't have this option
  291.         try:
  292.             # rebooting here makes sense if we run e.g. in qemu
  293.             return self.config.getboolean("NonInteractive","RealReboot")
  294.         except Exception, e:
  295.             logging.debug("no RealReboot found, returning false (%s) " % e)
  296.             return False
  297.     def error(self, summary, msg, extended_msg=None):
  298.         " display a error "
  299.         logging.error("%s %s (%s)" % (summary, msg, extended_msg))
  300.     def abort(self):
  301.         logging.error("view.abort called")
  302.  
  303.  
  304. if __name__ == "__main__":
  305.  
  306.   view = DistUpgradeViewNonInteractive()
  307.   fp = NonInteractiveFetchProgress()
  308.   ip = NonInteractiveInstallProgress()
  309.  
  310.   #ip.error("linux-image-2.6.17-10-generic","post-installation script failed")
  311.   ip.error("xserver-xorg","pre-installation script failed")
  312.  
  313.   cache = apt.Cache()
  314.   for pkg in sys.argv[1:]:
  315.     #if cache[pkg].isInstalled:
  316.     #  cache[pkg].markDelete()
  317.     #else:
  318.     cache[pkg].markInstall()
  319.   cache.commit(fp,ip)
  320.   time.sleep(2)
  321.   sys.exit(0)
  322.